# -------------------------- 问题增强 --------------------------
# 基于LLM更加准确的增加用户提问题的细节
from openai import OpenAI


def problem_enhancements(user_input, token):
    prompt = f"""
    你是一名专业的学术助手，擅长将简短主题句扩展为更详细的问题。

    示例
    输入：
    “人工智能在医疗影像中的应用研究。”

    输出：
    “我想查找关于人工智能（AI）在医疗影像领域的具体应用研究，特别是深度学习和机器学习技术在影像诊断、影像分割、疾病检测及预测分析中的具体应用与效果。这包括利用卷积神经网络（CNN）进行影像数据的自动分类与特征提取，生成对抗网络（GAN）用于图像增强与数据生成，以及迁移学习技术在有限数据集条件下的高效训练。此外，研究还应关注这些技术在X光、CT、MRI等医学影像类型中的应用表现，分析其对疾病早期诊断、治疗效果评估及临床辅助决策的实际价值与影响。”

    用户输入:{user_input}
    请你将用户输入转化一个更加详细的问题(只输出问题，不要输出任何其他内容，且只输出一个，且不换行)。
    """
    client = OpenAI(
        api_key=token,  # 此处传token，不带Bearer
        base_url="https://internlm-chat.intern-ai.org.cn/puyu/api/v1/",
    )
    chat_rsp = client.chat.completions.create(
        model="internlm2.5-latest",
        messages=[{"role": "user", "content": prompt}],
    )
    return chat_rsp.choices[0].message.content

if __name__ == "__main__":
    user_input = "脑机接口技术的未来应用前景如何？"
    token = "eyJ0eXBlIjVHSVNURVIiLCJpc3MiOiJPcGVuWExhYiIsImlhdCI6MTczNDM0NjQ4NSwiY2xpZW50SWQiOiJlYm1ydm9kNnlvMG5semFlazF5cCIsInBob25lIjoiMTg1MDg0MzY4OTMiLCJ1dWlkIjoiMTQ3MjkxNDUtNDcyZC00NGQzLWI4NWItNzgzNmY2OGQxMTY2IiwiZW1haWwiOiIiLCJleHAiOjE3NDk4OTg0ODV9.QX4S4jqXk3aI3_my_b5BEfTg02Rm8yLyYqabm8m4y5MYTDRO1lsf2_7hcABcXhMwoAbDC3izSvWIf5vZzwFvSQ"
    print(problem_enhancements(user_input, token))